from pyDes import des, PAD_PKCS5, ECB



class Encrypt:
    def __init__(self):
        self.DES_KEY = "00000000"

    def input_key(self):
        self.DES_KEY = input("请输入加密密钥")

    def encrypt(self, s):
        s = s.encode()  # 这里中文要转成字节
        des_obj = des(self.DES_KEY, ECB, self.DES_KEY, padmode=PAD_PKCS5)  # 初始化一个des对象，参数是秘钥，加密方式，偏移， 填充方式
        secret_bytes = des_obj.encrypt(s)  # 用对象的encrypt方法加密
        return secret_bytes.hex()
